package com.c2b.algorithm.leetcode.base;

import java.util.HashSet;
import java.util.Set;

/**
 * <a href='https://leetcode.cn/problems/find-the-distinct-difference-array/'>找出不同元素数目差数组(Find the Distinct Difference Array)</a>
 * <p>给你一个下标从 0 开始的数组 nums ，数组长度为 n 。</p>
 * <p>nums 的 不同元素数目差 数组可以用一个长度为 n 的数组 diff 表示，其中 diff[i] 等于前缀 nums[0, ..., i] 中不同元素的数目 减去 后缀 nums[i + 1, ..., n - 1] 中不同元素的数目。</p>
 * <p>返回 nums 的 不同元素数目差 数组。</p>
 * <p>注意 nums[i, ..., j] 表示 nums 的一个从下标 i 开始到下标 j 结束的子数组（包含下标 i 和 j 对应元素）。特别需要说明的是，如果 i > j ，则 nums[i, ..., j] 表示一个空子数组。</p>
 *
 * <p>
 * <b>示例</b>
 * <pre>
 * 示例 1：
 *      输入：nums = [1,2,3,4,5]
 *      输出：[-3,-1,1,3,5]
 *      解释：
 *          对于 i = 0，前缀中有 1 个不同的元素，而在后缀中有 4 个不同的元素。因此，diff[0] = 1 - 4 = -3 。
 *          对于 i = 1，前缀中有 2 个不同的元素，而在后缀中有 3 个不同的元素。因此，diff[1] = 2 - 3 = -1 。
 *          对于 i = 2，前缀中有 3 个不同的元素，而在后缀中有 2 个不同的元素。因此，diff[2] = 3 - 2 = 1 。
 *          对于 i = 3，前缀中有 4 个不同的元素，而在后缀中有 1 个不同的元素。因此，diff[3] = 4 - 1 = 3 。
 *          对于 i = 4，前缀中有 5 个不同的元素，而在后缀中有 0 个不同的元素。因此，diff[4] = 5 - 0 = 5 。
 *
 * 示例 2：
 *      输入：nums = [3,2,3,4,2]
 *      输出：[-2,-1,0,2,3]
 *      解释：
 *          对于 i = 0，前缀中有 1 个不同的元素，而在后缀中有 3 个不同的元素。因此，diff[0] = 1 - 3 = -2 。
 *          对于 i = 1，前缀中有 2 个不同的元素，而在后缀中有 3 个不同的元素。因此，diff[1] = 2 - 3 = -1 。
 *          对于 i = 2，前缀中有 2 个不同的元素，而在后缀中有 2 个不同的元素。因此，diff[2] = 2 - 2 = 0 。
 *          对于 i = 3，前缀中有 3 个不同的元素，而在后缀中有 1 个不同的元素。因此，diff[3] = 3 - 1 = 2 。
 *          对于 i = 4，前缀中有 3 个不同的元素，而在后缀中有 0 个不同的元素。因此，diff[4] = 3 - 0 = 3 。
 * </pre>
 * </p>
 *
 * <p>
 * <b>提示：</b>
 *  <ul>
 *      <li>1 <= n == nums.length <= 50</li>
 *      <li>1 <= nums[i] <= 50</li>
 *  </ul>
 * </p>
 *
 * @author c2b
 * @since 2024/1/23 13:36
 */
public class LC2670FindTheDistinctDifferenceArray_S {

    static class Solution {
        public int[] distinctDifferenceArray(int[] nums) {
            int n = nums.length;

            Set<Integer> set1 = new HashSet<>();
            Set<Integer> set2 = new HashSet<>();
            int[] preCnt = new int[n];
            for (int i = 0; i < n ; ++i) {
                set1.add(nums[i]);
                preCnt[i] = set1.size();
            }

            int[] sufCnt = new int[n];
            for (int i = n - 1; i >= 0; --i) {
                sufCnt[i] = set2.size();
                set2.add(nums[i]);
            }

            int[] diff = new int[n];
            for (int i = 0; i < n; i++) {
                diff[i] = preCnt[i] - sufCnt[i];
            }
            return diff;
        }
    }


    public static void main(String[] args) {
        Solution solution = new Solution();
        Printer.printArrayInt(solution.distinctDifferenceArray(new int[]{1, 2, 3, 4, 5}));
        Printer.printArrayInt(solution.distinctDifferenceArray(new int[]{3, 2, 3, 4, 2}));
        Printer.printArrayInt(solution.distinctDifferenceArray(new int[]{37,37,37,37,33}));
    }
}
